from pom.mainpage import MainPage

class MainModel:

    def __init__(self):
        self.mainPage = MainPage()

    def go_to_login_page(self):
        self.mainPage.login_link().click()

    def go_to_register_page(self):
        self.mainPage.register_link().click()
